// 初始化gorm
package models

import (
	"github.com/jinzhu/gorm"
	_ "github.com/jinzhu/gorm/dialects/sqlite"
	"os"
)

var db *gorm.DB

func init() {

	var err error
	if err = os.MkdirAll("data", 0777); err != nil {
		panic("创建文件夹data失败")
	}
	db, err = gorm.Open("sqlite3", "data/data.db")
	if err != nil {
		panic("连接数据库失败")
	}
	//defer db.Close()

	db.AutoMigrate(&User{}, &Note{})

	// 如果数据库中没有用户表，数据，那么我们新增一条记录
	var count int
	// 查询用户表数据
	if err := db.Model(&User{}).Count(&count).Error; err == nil && count == 0 {
		// gorm 的坑点，如果设置为0，就会取默认值。
		db.Create(&User{
			Name:   "admin",
			Email:  "admin@qq.com",
			Pwd:    "admin",
			Avatar: "/static/images/info-img.png",
			Role:   1,
		})
	}

}
